From pytorch/pytorch:latest
RUN apt-get update && \
    apt-get upgrade -y &&\
    apt-get install -y curl&&\
    apt-get install -y libgl1-mesa-glx&& \
    apt-get install -y libglib2.0-dev
ADD train_mnist.py .
ADD requirements.txt .
RUN pip install -i https://pypi.douban.com/simple/ -r requirements.txt
RUN pip install -i https://pypi.douban.com/simple/ opencv-python